"""
分组与聚合功能演示示例

该示例展示了Table类的按列分组和数据透视表创建等聚合功能。
"""
from ..dataframe import Table

# 分组与聚合演示
print("===== 分组与聚合功能演示 =====")

table = Table()
table.load_from_csv('../demo_data.csv')

# 按部门分组
groups = table.group_by('department')
print("各部门员工数量统计:")
for dept, dept_table in groups.items():
    print(f"- {dept}: {dept_table.get_row_count()} 人")

# 创建薪资统计透视表
pivot = table.pivot_table(values='salary', index='department', aggfunc=['mean', 'max'])
print("部门薪资统计透视表:\n", pivot)